for loop vs while loop in Python